package com.gmrz.uaf.servlet.v1.filter;

import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpServletResponseWrapper;
import java.io.IOException;
import java.io.PrintWriter;

/**
 * Created by gmrz-ops on 2017/7/12.
 */
public class UAFResponseWrapper extends HttpServletResponseWrapper {
    private UAFWriter uafWriter;


    public UAFResponseWrapper(HttpServletResponse response) {
        super(response);
    }

    @Override
    public PrintWriter getWriter() throws IOException {
        uafWriter = new UAFWriter(super.getWriter());
        return uafWriter;
    }



    public UAFWriter getUAFWriter() {
        return uafWriter;
    }

}
